import os
import sys

from experiment4 import Experiment4



def run_CLG():
    exp = Experiment4(model_type='gvsm', lang_code='zh', repo_path='chinese_only', use_translated_data=False,
                      term_similarity_type='cl_w')
    exp.run()


def run_WEG():
    exp = Experiment4(model_type='gvsm', lang_code='en', repo_path='chinese_only', use_translated_data=True,
                      term_similarity_type='gensim_wv')
    exp.run()


def run_VSM1():
    exp = Experiment4(data_dir='D:\\ACode\\EMSE', model_type='vsm', lang_code='en', repo_path='arthas', use_translated_data=True,
                      term_similarity_type='cl_w')
    exp.run()

def run_VSM2():
    exp = Experiment4(data_dir='C:\\Users\\LeiYu\\Desktop', model_type='vsm', lang_code='zh', repo_path='xLua', use_translated_data=True,
                      term_similarity_type='cl_w')
    exp.run()



if __name__ == "__main__":
    run_VSM1()

